昨天我們已經申請到 API Key,今天教大家怎麼成功呼叫 OpenAI API。
再來做一個最簡單的 對話機器人 輸入問題,得到回覆。
新建一個 .py
檔案,並加上以下程式碼:
import os
from openai import OpenAI
from dotenv import load_dotenv
# 載入環境變數
load_dotenv()
client = OpenAI(api_key=os.getenv("OPENAI_API_KEY"))
while True:
user_input = input("你:")
# 如果輸入 exit 就離開
if user_input.lower() in ["exit", "quit", "bye"]:
print("機器人:下次見!")
break
response = client.chat.completions.create(
model="gpt-4o-mini", # 選擇模型
messages=[
{"role": "system", "content": "你是一個友善的小助理。"},
{"role": "user", "content": user_input}
]
)
print("機器人:", response.choices[0].message.content)
來看看成果:
今天我們完成了一個最基礎的「聊天機器人」:
能持續輸入問題
AI 會回覆答案
明天的 Day 4,我們要更深入教 AI角色扮演,
讓它不只是單純聊天 而是可以「變身」成不同身份的專家!